package com.zjst.hashtable;

import java.util.Scanner;

/**
 * @author zhanb
 * @Date 2020/5/21
 */
public class HashTableDemo {
    public static void main(String[] args) {
        //创建哈希表
        HashTab hashTab = new HashTab(7);
        //写一个简单的菜单
        String key = "";
        Scanner scanner = new Scanner(System.in);
        while(true) {
            System.out.println("add:  添加雇员");
            System.out.println("list: 显示雇员");
            System.out.println("find: 查找雇员");
            System.out.println("delete: 删除雇员");
            System.out.println("exit: 退出系统");

            key = scanner.next();
            switch (key) {
                case "add":
                    System.out.println("输入id");
                    int id = scanner.nextInt();
                    System.out.println("输入名字");
                    String name = scanner.next();
                    //创建 雇员
                    Emp emp = new Emp(id, name);
                    hashTab.add(emp);
                    break;
                case "list":
                    hashTab.list();
                    break;
                case "delete":
                    System.out.println("请输入要删除的id");
                    id = scanner.nextInt();
                    hashTab.deleteEmpById(id);
                    break;
                case "find":
                    System.out.println("请输入要查找的id");
                    id = scanner.nextInt();
                    hashTab.findEmpById(id);
                    break;
                case "exit":
                    scanner.close();
                    System.exit(0);
                default:
                    break;
            }
        }

    }
}

/**
 * 创建HashTab 管理多条链表
 */
class HashTab {
    private EmpLinkedList[] empLinkedListArray;
    //表示有多少条链表
    private int size;


    /**
     * 构造器
     * @param size
     */
    public HashTab(int size) {
        this.size = size;
        //初始化empLinkedListArray
        empLinkedListArray = new EmpLinkedList[size];
        //？留一个坑, 这时不要分别初始化每个链表
        for(int i = 0; i < size; i++) {
            empLinkedListArray[i] = new EmpLinkedList();
        }
    }

    /**
     * 添加雇员
     * @param emp
     */
    public void add(Emp emp) {
        //根据员工的id ,得到该员工应当添加到哪条链表
        int empLinkedListNO = hashFun(emp.id);
        //将emp 添加到对应的链表中
        //empLinkedListArray[empLinkedListNO].add(emp)
        empLinkedListArray[empLinkedListNO].addByOrder(emp);

    }
    //遍历所有的链表,遍历hashtab
    public void list() {
        for(int i = 0; i < size; i++) {
            empLinkedListArray[i].list(i);
        }
    }
    /**
     * 根据输入的id,查找雇员
     * @param id
     */
    public void findEmpById(int id) {
        //使用散列函数确定到哪条链表查找
        int empLinkedListNO = hashFun(id);
        Emp emp = empLinkedListArray[empLinkedListNO].findEmpById(id);
        //找到
        if(emp != null) {
            System.out.printf("在第%d条链表中找到 雇员 id = %d\n", (empLinkedListNO + 1), id);
        }else{
            System.out.println("在哈希表中，没有找到该雇员~");
        }
    }

    /**
     * 删除用户
     * @param id
     */
    public void deleteEmpById(int id){
        //使用散列函数确定到哪条链表查找
        int empLinkedListNO = hashFun(id);
        empLinkedListArray[empLinkedListNO].deleteEmpById(id);
    }

    /**
     * 编写散列函数, 使用一个简单取模法
     * @param id
     * @return
     */
    public int hashFun(int id) {
        return id % size;
    }


}

//表示一个雇员
class Emp {
    public int id;
    public String name;
    //next 默认为 null
    public Emp next;
    public Emp(int id, String name) {
        super();
        this.id = id;
        this.name = name;
    }
}

/**
 * 创建EmpLinkedList ,表示链表
 */
class EmpLinkedList {
    /**
     * 头指针，执行第一个Emp,因此我们这个链表的head 是直接指向第一个Emp
     */
    private Emp head;



    /**
     *  //添加雇员到链表
     *     //说明
     *     //1. 假定，当添加雇员时，id 是自增长，即id的分配总是从小到大
     *     //   因此我们将该雇员直接加入到本链表的最后即可
     * @param emp
     */
    public void add(Emp emp) {
        //如果是添加第一个雇员
        if (head == null) {
            head = emp;
            return;
        }
        //如果不是第一个雇员，则使用一个辅助的指针，帮助定位到最后
        Emp curEmp = head;
        while (true) {
            if (curEmp.next == null) {//说明到链表最后
                break;
            }
            curEmp = curEmp.next; //后移
        }
        //退出时直接将emp 加入链表
        curEmp.next = emp;
    }

    /**
     * 添加用户id通过顺序
     * @param emp
     */
    public void addByOrder(Emp emp){
        if (head == null){
            head = emp;
            return;
        }
        //如果链表只有一个节点
        if (head.next ==null){
            //如果比第一个节点小，直接头插
            if (head.id >emp.id){
                emp.next =head;
                head =emp;
            }else {
                head.next = emp;
            }
            return;
        }
        //链表不止一个节点
        Emp cur = head;
        boolean flag  = false;
        while (true){
            //链表的最后一个
            if (cur.next == null) {
                break;
            }
            //找到插入的前一个节点
            if (cur.next.id > emp.id){
                flag = true;
                break;
            }
            cur = cur.next;
        }
        //找到插入的节点
      if (flag){
          emp.next = cur.next;
          cur.next = emp;
      }else {   //如果前面的编号都比他小直接加入最后
          cur.next =emp;
      }
    }


    /**
     * 遍历链表的雇员信息
     * @param no
     */
    public void list(int no) {
        //说明链表为空
        if (head == null) {
            System.out.println("第 " + (no + 1) + " 链表为空");
            return;
        }
        System.out.print("第 " + (no + 1) + " 链表的信息为");
        //辅助指针
        Emp curEmp = head;
        while (true) {
            System.out.printf(" => id=%d name=%s\t", curEmp.id, curEmp.name);
            //说明curEmp已经是最后结点
            if (curEmp.next == null) {
                break;
            }
            //后移，遍历
            curEmp = curEmp.next;
        }
        System.out.println();
    }


    /**根据id查找雇员
     * 如果查找到，就返回Emp, 如果没有找到，就返回null
     * @param id
     * @return
     */
    public Emp findEmpById(int id) {
        //判断链表是否为空
        if (head == null) {
            System.out.println("链表为空");
            return null;
        }
        //辅助指针
        Emp curEmp = head;
        while (true) {
            //找到,这时curEmp就指向要查找的雇员
            if (curEmp.id == id) {
                break;
            }
            //退出
            //说明遍历当前链表没有找到该雇员
            if (curEmp.next == null) {
                curEmp = null;
                break;
            }
            //以后
            curEmp = curEmp.next;
        }
        return curEmp;
    }

    /**
     * 删除用户
     * @param id 用户id
     */
    public void deleteEmpById(int id){
        Emp cur = head;
        if (head == null) {
            return;
        }
        if(head.id == id){
            head =head.next;
            return;
        }
        boolean flag =false;
        while (cur.next != null){
            if (cur.next.id ==id){
                flag =true;
                break;
            }
            cur = cur.next;
        }
        if (flag){
            cur.next = cur.next.next;
        }
    }
}
